package topK;

// topK问题：目前假设求前k大！

import java.util.Comparator;
import java.util.PriorityQueue;

public class Test {
    // 方法一：放入相同堆中，出k次
    public void topK1(int[] arr,int k) {
        if(k==0) {
            return;
        }
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1; // 大根堆
            }
        });
        // 放入堆中
        for (int i = 0; i < arr.length; i++) {
            priorityQueue.offer(arr[i]);
        }
        // 出k次
        for (int i = 0; i < k; i++) {
            System.out.print(priorityQueue.poll() + " ");
        }
    }

    // 方法二：k大小的相反堆 + 遍历比较出堆
    public int[] topK2(int[] arr, int k) {
        int[] ret = new int[k];
        if (k==0) {
            return ret;
        }
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(k,new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o1-o2; // 小根堆
            }
        });
        // 建立k大小的小根堆（数组元素进行存入）
        for (int i = 0; i < arr.length; i++) {
            if(priorityQueue.size()<k) {
                priorityQueue.offer(arr[i]);
            } else {
                // 说明已经建好了k堆，要进行比较变换
                // 获取栈顶元素
                int top = priorityQueue.peek();
                // 栈顶元素与数组遍历的i下标的元素去比较大小
                // 找前k个最大的
                if(arr[i] > top) {
                    // 要是新元素大,则说明堆顶元素不包含在内
                    priorityQueue.poll();
                    priorityQueue.offer(arr[i]);
                }
            }
        }
        // 然后进行出k个元素：
        for (int i = 0; i < k; i++) {
            int val = priorityQueue.poll();
            ret[i] = val;
        }
        return ret;
    }

    // 实现最小k个数
    // 另外参考力扣写第三种方法：
    public int[] topK3(int[] arr, int k) {
        // 先建立k大小的数组存入前k个元素
        int[] ret = new int[k];
        if(k==0) {
            return ret;
        }
        // 然后建立k大小优先队列，要实现比较器
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1; // 大堆
            }
        });
        // 建立k大小的大根堆：放入数据
        for (int i = 0; i < k; i++) {
            priorityQueue.offer(arr[i]);
        }
        // 从k下标开始进行比较并变换
        for (int i = k; i < arr.length; i++) {
            if(arr[i] < priorityQueue.peek()) {
                // 如果原来的大就换掉
                priorityQueue.poll();
                priorityQueue.offer(arr[i]);
            }
        }
        // 然后弹出前k个数据存储在数组中
        for (int i = 0; i < k; i++) {
            ret[i] = priorityQueue.poll();
        }
        return ret;
    }
}
